package mq;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;

import java.util.LinkedList;
import java.util.concurrent.TimeUnit;

/**
 * @author： pang-yun
 * @date： 2021-11-02 09:49
 * 功能目的 ：用于
 */

@Slf4j
public class Test21 {
    public static void main(String[] args) {
        MessageQueue queue = new MessageQueue(2);

        for (int i=0; i<3; i++){
            int id =i;
            new Thread(()->{
                queue.put(new Message(id,"------value: "+id));
            },"生产者"+id).start();
        }

        new Thread(()->{
          while (true){
              try {
                  TimeUnit.SECONDS.sleep(1);
              } catch (InterruptedException e) {
                  e.printStackTrace();
              }
              queue.take();
          }
        },"消费者").start();
    }
}


//消息队列类  java 线程之间通信
@Slf4j
class MessageQueue {
    private LinkedList<Message> list = new LinkedList<>();
    private int capacity;

    public MessageQueue(int capacity) {
        this.capacity = capacity;
    }

    // 获取消息
    public Message take() {
        synchronized (list) {
            // 首先判断 是否为空 空则不取
            while (list.isEmpty()) {
                try {
                    log.info("队列为空，消费者线程等待");
                    list.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            // 从 队列 头部 获取消息 并返回
            Message message = list.removeFirst();
            list.notifyAll();  // 唤醒 线程  告诉已经取走一个  看看是否不满了
            log.info("已消费消息: {}", message);
            return message;
        }
    }

    // 放入消息
    public void put(Message message) {
        synchronized (list) {
            while (list.size() == capacity) {
                try {
                    log.info("队列已满，生产者线程等待!");
                    list.wait();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
            // 在尾部放入消息
            list.addLast(message);
            list.notifyAll();
            log.info("已生产消息: {}", message);
        }
    }
}


// 要传递的消息  final  修饰  防止重写
@Data
@AllArgsConstructor
final class Message {
    private int id;
    private Object value;
}
